'''
Assignment: Create PT Practice #4 (from Display3)
This version will play again
'''


from codex import *
from time import sleep


messages = ["Press Up", "Press Down", "Press Left", "Press Right",
            "Press A", "Press B", "Press Left", "Press Down"]
btns = [BTN_U, BTN_D, BTN_L, BTN_R, 
        BTN_A, BTN_B, BTN_L, BTN_D]


delay = 1
count = 0


# One function for game play
def play_game():
    global count
    for ind in range(len(messages)):
        message = messages[ind]
        button = btns[ind]
        display.show(message)
        sleep(delay)


        pressed = buttons.is_pressed(button)
        if pressed:
            pixels.set(ind%4, GREEN)
            count = count + 1
        else:
            pixels.set(ind%4, RED)
        sleep(delay)
        pixels.set(ind%4, BLACK)


def results(count):
    # turn off all pixels and clear screen
    for pix in range(4):
        pixels.set(pix, BLACK)
    display.clear()


    if count == len(btns):
        end_message = "You won!"
        col = GREEN
    elif count == 0:
        end_message = "You lost"
        col = RED
    else:
        end_message = "Keep trying"
        col = BLUE
    
    display.draw_text(end_message, scale=3, x=30, y=100, color=col)
    # running pixel lights
    for num in range(30):
        pixels.set(num%4, col)
        sleep(0.2)
        pixels.set(num%4, BLACK)
    


def play_again():
    global continues, count
    display.clear()
    display.print("Play again?")
    display.print("A = Yes")
    display.print("B = No")
    while True:
        if buttons.is_pressed(BTN_A):
            pixels.set([BLACK, BLACK, BLACK, BLACK])
            display.clear()
            count = 0
            break
        if buttons.is_pressed(BTN_B):
            continues = False
            break




# Main Program
continues = True
while continues:
    play_game()
    results(count)
    play_again()


display.clear()
display.print("Game Over")
display.print("Thank you")